package com.zy.explore.sys.interfaces.assembler;

import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.zy.explore.common.infrastructure.util.DataBeanUtil;
import com.zy.explore.common.interfaces.frm.CommonIntIdFrm;
import com.zy.explore.common.interfaces.vo.PageVO;
import com.zy.explore.sys.application.service.SysUserService;
import com.zy.explore.sys.domain.mybatis.entity.SysMenu;
import com.zy.explore.sys.domain.mybatis.entity.SysRole;
import com.zy.explore.sys.domain.mybatis.entity.SysUser;
import com.zy.explore.sys.infrastructure.constant.SysModuleConstant;
import com.zy.explore.sys.interfaces.frm.CheckPwdFrm;
import com.zy.explore.sys.interfaces.frm.ModifyPwdFrm;
import com.zy.explore.sys.interfaces.frm.SysUserFrm;
import com.zy.explore.sys.interfaces.vo.LastLoginInfoVO;
import com.zy.explore.sys.interfaces.vo.SysUserVO;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageImpl;
import org.springframework.data.domain.PageRequest;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import tk.mybatis.mapper.entity.Example;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.List;

/**
 * 系统用户Assembler
 *
 * @author ChinaYin  主页:https://gitee.com/zy-explorej2ee 邮箱:zyexplorej2ee@sina.com
 * @date 2021年06月13日 12:13:50
 */
@Component
public class SysUserAssembler {

    /**
     * 注入service
     */
    @Resource
    private SysUserService service;

    /**
     * 新增数据
     *
     * @param frm 参数对象
     * @return 视图对象
     */
    @Transactional(rollbackFor = Throwable.class)
    public SysUserVO insert(SysUserFrm frm) {
        SysUser entity = convertFromFrm(frm);
        entity.setCreateTime(new Date());
        SysUser res = service.insertSelective(entity);
        if (res != null) {
            // 新增用户角色
            service.addRole(Arrays.asList(res.getId()), Arrays.asList(SysModuleConstant.NORMAL_USER_ROLE));
            return convertToVO(res);
        }
        return null;
    }

    /**
     * 删除对象
     *
     * @param frm 参数对象
     * @return true:删除成功 false:删除失败
     */
    public Integer delete(CommonIntIdFrm frm) {
        return service.deleteByPrimaryKey(frm.getId());
    }

    /**
     * 更新对象
     *
     * @param frm 待更新的Frm对象
     * @return 成功返回更新后的VO对象，失败返回null
     */
    @Transactional(rollbackFor = Throwable.class)
    public SysUserVO update(SysUserFrm frm) {
        if (frm.getId() == null) {
            return null;
        }
        SysUser entity = convertFromFrm(frm);
        entity.setUpdateTime(new Date());
        service.addSubAdmin(entity.getId(), frm.getSubAdmin());
        int res = service.updateByPrimaryKeySelective(entity);
        if (res == 1) {
            return convertToVO(entity);
        }
        return null;
    }

    /**
     * 根据ID查询数据
     *
     * @param frm 主键Frm对象
     * @return 成功返回对应ID的VO对象，失败返回null
     */
    public SysUserVO findById(CommonIntIdFrm frm) {
        SysUser entity = service.getByPrimaryKey(frm.getId());
        return convertToVO(entity);
    }

    /**
     * 根据条件查询数据集合
     *
     * @param frm 查询条件对象
     * @return 满足条件的数据集合
     */
    public List<SysUserVO> getList(SysUserFrm frm) {
        Example example = SysUser.getExample(frm);
        List<SysUser> list = service.selectByExample(example);
        if (!CollectionUtils.isEmpty(list)) {
            List<SysUserVO> retList = new ArrayList<>();
            // 系统管理员不返回给前端
            list.stream().filter(x -> x.getId() > 1).forEach(x -> retList.add(convertToVO(x)));
            return retList;
        }
        return new ArrayList<>();
    }

    /**
     * 根据条件查询分页
     *
     * @param frm 查询条件
     * @return 满足条件的分页
     */
    public PageVO<SysUserVO> getPageInfo(SysUserFrm frm) {
        Example example = SysUser.getExample(frm);
        PageInfo<SysUser> pageInfo = PageHelper.startPage(frm.getPageNum(), frm.getPageSize()).doSelectPageInfo(() -> service.selectByExample(example));
        PageRequest pageRequest = PageRequest.of(frm.getPageNum() - 1, frm.getPageSize());
        Page<SysUserVO> retPage = new PageImpl<>(new ArrayList<>(), pageRequest, 0);
        if (pageInfo == null || CollectionUtils.isEmpty(pageInfo.getList())) {
            return new PageVO<>(retPage);
        }
        List<SysUserVO> retList = new ArrayList<>();
        pageInfo.getList().forEach(x -> retList.add(convertToVO(x)));
        retPage = new PageImpl<>(retList, pageRequest, pageInfo.getTotal());
        return new PageVO<>(retPage);
    }

    /**
     * 根据用户名查询用户信息
     *
     * @param username 用户名
     * @return {@link SysUser} 返回用户对象
     */
    public SysUser getUserByUsername(String username) {
        return service.getUserByUsername(username);
    }

    /**
     * 获取用户角色列表
     *
     * @param id 用户id
     * @return {@link List<SysRole>} 用户角色集合
     */
    public List<SysRole> getUserRoleList(Integer id) {
        return service.getUserRoleList(id);
    }

    /**
     * 获取用户菜单列表
     *
     * @param id 用户id
     * @return {@link List<SysMenu>} 用户菜单列表
     */
    public List<SysMenu> getUserMenuList(Integer id) {
        return service.getUserMenuList(id);
    }

    /**
     * 获取用户最后的登录信息
     *
     * @param id 用户id
     * @return {@link LastLoginInfoVO} 登录信息VO
     */
    public LastLoginInfoVO getLastLoginInfo(Integer id) {
        return service.getLastLoginInfo(id);
    }

    /**
     * 校验用户密码是否正确
     *
     * @param frm 检验参数
     * @return 返回1:正确 0:不正确
     */
    public Integer checkPwd(CheckPwdFrm frm) {
        return service.checkPwd(frm);
    }

    /**
     * 修改密码
     *
     * @param frm 修改密码参数
     * @return 返回1:成功 0:失败
     */
    public Integer modifyPwd(ModifyPwdFrm frm) {
        return service.modifyPwd(frm);
    }

    /**
     * 超管重置员工密码
     *
     * @param frm 员工ID对象
     * @return {@link Boolean}
     */
    public Boolean resetEmpPwd(CommonIntIdFrm frm) {
        return service.resetEmpPwd(frm.getId());
    }

    /**
     * frm转实体类
     *
     * @param frm 参数对象
     * @return 转换后的实体类对象
     */
    public static SysUser convertFromFrm(SysUserFrm frm) {
        return DataBeanUtil.copyOf(frm, SysUser.class, true);
    }

    /**
     * 实体类转VO
     *
     * @param entity 实体类
     * @return 转换后的对象
     */
    public static SysUserVO convertToVO(SysUser entity) {
       return entity == null ? null : DataBeanUtil.copyOf(entity, SysUserVO.class, true);
    }
}
